#!/usr/bin/python3
# coding=utf-8
#
# Copyright (C) 2023-2024. Huawei Technologies Co., Ltd. All rights reserved.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
# ===============================================================================

import numpy as np
import os


def gen_golden_data():
    M = 16*16*2
    K = 16*16*2


    input_a = np.tile(np.arange(1, K+1, dtype=np.float16), (M, 1))
    # input_a = (np.arange(1, K+1, dtype=np.float16)[None, :]
    #        + np.arange(M, dtype=np.float16)[:, None])
    # input_a = np.ones((M, K), dtype=np.float16)
    # 生成全1矩阵
    # input_a = np.arange(0, M*K, dtype=np.float16).reshape(M, K)

    os.system("mkdir -p input")
    os.system("mkdir -p output")
    input_a.tofile("./input/x1_gm.bin")



if __name__ == "__main__":
    gen_golden_data()
